package com.hjzd.ai.smartpractice.entity.DTO.chat;

import java.util.List;

/**
 * @author pengyonglei
 * @version 1.0.0
 */
public class ChatCompletionRequest {

	private List<ChatCompletionMessage> messages;
	private int max_tokens = 512;
	private boolean stream;
	private String model = "gpt-3.5-turbo";

	private String session_id;
	private String character;

	public String getCharacter() {
		return character;
	}

	public void setCharacter(String character) {
		this.character = character;
	}

	public ChatCompletionRequest() {
	}

	public ChatCompletionRequest(List<ChatCompletionMessage> messages, int max_tokens, boolean stream, String model) {
		this.messages = messages;
		this.max_tokens = max_tokens;
		this.stream = stream;
		this.model = model;
	}

	public List<ChatCompletionMessage> getMessages() {
		return messages;
	}

	public void setMessages(List<ChatCompletionMessage> messages) {
		this.messages = messages;
	}

	public int getMax_tokens() {
		return max_tokens;
	}

	public void setMax_tokens(int max_tokens) {
		this.max_tokens = max_tokens;
	}

	public boolean isStream() {
		return stream;
	}

	public void setStream(boolean stream) {
		this.stream = stream;
	}

	public String getModel() {
		return model;
	}

	public void setModel(String model) {
		this.model = model;
	}

	public String getSession_id() {
		return session_id;
	}

	public void setSession_id(String session_id) {
		this.session_id = session_id;
	}

	@Override
	public String toString() {
		return "ChatCompletionRequest{" +
				"messages=" + messages +
				", max_tokens=" + max_tokens +
				", stream=" + stream +
				", model='" + model + '\'' +
				'}';
	}
}
